import numpy as np
import  numpy as pd
import matplotlib.pyplot as plt
#导入数据
data = np.loadtxt('d:/dataset/Ad_sales.csv',skiprows=1,usecols=[1,2,3,4],delimiter=',')
print(data)
#对数据进行转置处理
data_T = data.T
print(data_T)
#设置Matplotlib正常显示中文和负号
plt.rcParams['font.sans-serif']=['SimHei']   #用黑体显示中文
plt.rcParams['axes.unicode_minus']=False  #正常显示负号
#创建绘图对象，并设置对象的宽度和高度
plt.figure(figsize=(8,4))
#绘制TV广告投入与销售额的散点图
plt.scatter(data_T[3],data_T[0],c='r',marker='o')
#绘制radio广告投入与销售额的散点图
plt.scatter(data_T[3],data_T[1],c='g',marker='x')
#绘制newspaper广告投入与销售额的散点图
plt.scatter(data_T[3],data_T[2],c='b',marker='v')

plt.xlabel('销售额')
plt.ylabel('广告投入')
plt.ylim(0,300)
plt.xlim(0.30)

#设置主标题、副标题
plt.suptitle('广告投入与与销售额之间的关系')
plt.title('单位:万元',fontsize=10,loc='right')
#添加图例
plt.legend(['TV','radio','newspaper'])
plt.savefig('d:/image/task4-19.png')  #保存图片
plt.show()